package com.htzos.common.util;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;
import gui.ava.html.image.generator.HtmlImageGenerator;


public class PrintscreenUtil{
	/**
	 * 将html页面转化为图片
	 * @param html html文件的节点字符串
	 * @param path	图片保存路径
	 * @param fileName	图片名称
	 * @param fileType	图片类型
	 * @return
	 */
	public static boolean createImageAsHtml(String html,String path,String fileName,String fileType){
		try{
			HtmlImageGenerator imageGenerator = new HtmlImageGenerator();
			imageGenerator.loadHtml(html);
//			imageGenerator.loadUrl("http://www.baidu.com");
			imageGenerator.saveAsImage(path+fileName+"."+fileType);
			return true;
		}catch(Exception e){
			return false;
		}
//		imageGenerator.saveAsHtmlWithMap("hello-world.html", "hello-world.png");
	}
	/**
	 * 将图片文件转化为字节数组字符串，并对其进行Base64编码处理
	 * @param imgFilePath 图片路径
	 * @return
	 */
	public static String GetImageStr(String imgFilePath) {
		byte[] data = null;

		// 读取图片字节数组
		try {
			InputStream in = new FileInputStream(imgFilePath);
			data = new byte[in.available()];
			in.read(data);
			in.close();
		} catch (IOException e) {
			e.printStackTrace();
		}

		// 对字节数组Base64编码
		BASE64Encoder encoder = new BASE64Encoder();
		return encoder.encode(data);// 返回Base64编码过的字节数组字符串
	}
	/**
	 * 对字节数组字符串进行Base64解码并生成图片
	 * @param imgStr 64位编码
	 * @param imgFilePath	生成的图片存放路径
	 * @return
	 */
	public static boolean GenerateImage(String imgStr, String imgFilePath) {
		if (imgStr == null){// 图像数据为空
			return false;
		} 
		BASE64Decoder decoder = new BASE64Decoder();
		try {
			// Base64解码
			byte[] bytes = decoder.decodeBuffer(imgStr);
			for (int i = 0; i < bytes.length; ++i) {
				if (bytes[i] < 0) {// 调整异常数据
					bytes[i] +=256;
				}
			}
			// 生成jpeg图片
			OutputStream out = new FileOutputStream(imgFilePath);
			out.write(bytes);
			out.flush();
			out.close();
			return true;
		} catch (Exception e) {
			return false;
		}
	}
}

